
from dataprocessor.NiiDataFormater import NiiDataProcessor
from dataprocessor.preregistrator import Registrator
from dataprocessor.SampleSplitter import split
if __name__=="__main__":
    pre = NiiDataProcessor("E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image",
                           "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label")
    pre.preprocess_itk()
    pre = NiiDataProcessor("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image",
                           "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label")
    pre.preprocess_itk()
    reg=Registrator("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image_result",
                    "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label_result",
                    "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image_result",
                    "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label_result")
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image_result_pre_reg_test",5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image_result_pre_reg_test", 5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label_result_pre_reg_test", 5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label_result_pre_reg_test", 5)